package com.xxx.multids;

/**
 * 当前数据源处理工具
 */
public class DataSourceHandler {

    /**
     * 保存当前数据源
     */
    private final static ThreadLocal<Integer> currentDataSource = new ThreadLocal<>();

    private DataSourceHandler() {
    }

    /**
     * 设置当前数据源
     *
     * @param datasource
     */
    public static void putDataSource(Integer datasource) {
        currentDataSource.set(datasource);
    }

    /**
     * 获取当前数据源
     *
     * @return
     */
    public static Integer getDataSource() {
        return currentDataSource.get();
    }

    /**
     * 清除当前数据源,恢复默认设置
     */
    public static void clear() {
        currentDataSource.remove();
    }
}
